python assign multiple variables in for loop